from shuangti.Item01.AIAssistant import AIAssistant


class TextGenerationAssistant(AIAssistant):
    """文本生成助手子类，支持的文本类型为自定义属性"""
    def __init__(self, aid: str, name: str, version: str, is_online: bool, ability_score: int, supported_text_types: str):
        super().__init__(aid, name, "文本助手", version, is_online, ability_score)
        self.supported_text_types = supported_text_types  # 自定义属性：支持的文本类型（诗歌、文案等）

    def execute_task(self, task_content: str) -> str:
        """执行文本生成任务（模拟诗歌生成）"""
        if not self.is_online:
            return f"❌ {self.name} 当前离线，无法执行任务。"
        # 模拟诗歌生成结果
        poem_example = "\n春眠不觉晓，处处闻啼鸟。\n夜来风雨声，花落知多少。"
        return (f"✅ {self.name}（文本助手）已执行任务：\n任务内容：{task_content}\n"
                f"处理结果：{poem_example}\n支持文本类型：{self.supported_text_types}")

    def __str__(self) -> str:
        base_info = super().__str__()
        return f"{base_info}, 支持文本类型: {self.supported_text_types}"